package com.qidianit.mes.masterdata.controller;

import com.qidianit.mes.common.pojo.MesResult;
import com.qidianit.mes.masterdata.service.UnitService;
import com.qidianit.mes.pojo.Unit;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.List;

/**
 * @company 起点编程
 * @author 戴着假发的程序员
 * @date 2025/7/22
 * @description 单位Controller
 */
@RestController
@RequestMapping("/units")
public class UnitController {

    @Autowired
    private UnitService unitService;

    /**
     * 查询所有单位
     */
    @GetMapping
    public MesResult<List<Unit>> getAllUnits() {
        return MesResult.success(unitService.list());
    }

    /**
     * 根据ID查询单位
     */
    @GetMapping("/{id}")
    public MesResult<Unit> getUnitById(@PathVariable Long id) {
        return MesResult.success(unitService.getById(id));
    }

    /**
     * 新增单位
     */
    @PostMapping
    public MesResult<Boolean> createUnit(@RequestBody Unit unit) {
        return MesResult.success(unitService.save(unit));
    }

    /**
     * 更新单位
     */
    @PutMapping("/{id}")
    public MesResult<Boolean> updateUnit(@PathVariable Long id, @RequestBody Unit unit) {
        unit.setUnitId(id);
        return MesResult.success(unitService.updateById(unit));
    }

    /**
     * 删除单位
     */
    @DeleteMapping("/{id}")
    public MesResult<Boolean> deleteUnit(@PathVariable Long id) {
        return MesResult.success(unitService.removeById(id));
    }
}